from pydantic import BaseModel

class ModelInput(BaseModel):
    user_input: str
    session_id: int

class BaseModelInput(BaseModel):
    user_input: str

class GenerateXhs(BaseModel):
    user_input: str